package com.ruoyi.eqmonitor.utils;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.IdUtil;
import com.jacob.activeX.ActiveXComponent;
import com.jacob.com.Dispatch;
import com.jacob.com.Variant;
import com.ruoyi.common.core.exception.ServiceException;
import lombok.extern.slf4j.Slf4j;

import java.io.*;

/**
 * TTS工具类
 *
 * 参考jacob文档, 需要将jacob-1.18-x64.dll放入jdk的bin目录下,仅限windows
 */
@Slf4j
public class TTSUtil {
    /** 文件前缀 */
    public static String FILE_PREFIX = null;

    static {
        String rootPath = System.getProperty("user.dir");
        FILE_PREFIX = rootPath + File.separator + "tts" + File.separator;
        // 文件夹不存在转语音时会失败, 所以
        FileUtil.mkdir(FILE_PREFIX);
    }

    /**
     * 语音转文字并播放
     *
     * @param text 需要转语音的文本
     */
    public static File textToSpeechFile(String text) {
        ActiveXComponent ax = null;
        try {
            ax = new ActiveXComponent("Sapi.SpVoice");

            // 运行时输出语音内容
            Dispatch spVoice = ax.getObject();
            // 音量 0-100
            ax.setProperty("Volume", new Variant(100));
            // 语音朗读速度 -10 到 +10
            ax.setProperty("Rate", new Variant(-2));
            // 执行朗读
//            Dispatch.call(spVoice, "Speak", new Variant(text));

            // 下面是构建文件流把生成语音文件

            ax = new ActiveXComponent("Sapi.SpFileStream");
            Dispatch spFileStream = ax.getObject();

            ax = new ActiveXComponent("Sapi.SpAudioFormat");
            Dispatch spAudioFormat = ax.getObject();

            // 设置音频流格式
            Dispatch.put(spAudioFormat, "Type", new Variant(22));
            // 设置文件输出流格式
            Dispatch.putRef(spFileStream, "Format", spAudioFormat);
            // 调用输出 文件流打开方法，创建一个文件   支持 .wav  .mp3 这两个是我尝试的，都支持，和一些个听书读的差不多
            String fullPath = FILE_PREFIX + IdUtil.fastSimpleUUID() +".mp3";
            Dispatch.call(spFileStream, "Open", new Variant(fullPath), new Variant(3), new Variant(true));
            // 设置声音对象的音频输出流为输出文件对象
            Dispatch.putRef(spVoice, "AudioOutputStream", spFileStream);
            // 设置音量 0到100
            Dispatch.put(spVoice, "Volume", new Variant(100));
            // 设置朗读速度
            Dispatch.put(spVoice, "Rate", new Variant(-2));
            // 开始朗读
            Dispatch.call(spVoice, "Speak", new Variant(text));

            // 关闭输出文件
            Dispatch.call(spFileStream, "Close");
            Dispatch.putRef(spVoice, "AudioOutputStream", null);

            spAudioFormat.safeRelease();
            spFileStream.safeRelease();
            spVoice.safeRelease();
            ax.safeRelease();

            return new File(fullPath);
        } catch (Exception e) {
            log.error("文本转语音失败: {}", text, e);
            throw new ServiceException("文字转语音失败");
        }
    }

    /**
     * 语音转文字并播放
     *
     * @param text 需要转语音的文本
     */
    public static byte[] textToSpeechByteArray(String text) {
        // 文字转语音
        File file = textToSpeechFile(text);

        try (
                FileInputStream fileInputStream = IoUtil.toStream(file);
                ByteArrayOutputStream outputStream = new ByteArrayOutputStream()
        ) {
            // 文件转ByteArray
            IoUtil.copy(fileInputStream, outputStream);
            return outputStream.toByteArray();
        } catch (IOException e) {
            log.error("文本转语音失败: {}", text, e);
            throw new ServiceException("文字转语音失败");
        } finally {
            // 删除文件
            FileUtil.del(file);
        }
    }

    /**
     * 语音转文字并播放
     *
     * @param text 需要转语音的文本
     */
    public static InputStream textToSpeechStream(String text) {
        return IoUtil.toStream(textToSpeechByteArray(text));
    }
}
